package com.xxl.job.admin.controller;

import cn.hutool.core.lang.Validator;
import cn.hutool.core.util.StrUtil;
import com.xxl.job.admin.core.exception.XxlJobException;
import com.xxl.job.admin.core.model.XxlJobGroup;
import com.xxl.job.admin.core.model.XxlJobInfo;
import com.xxl.job.admin.core.model.XxlJobLog;
import com.xxl.job.admin.core.model.XxlJobUser;
import com.xxl.job.admin.core.route.ExecutorRouteStrategyEnum;
import com.xxl.job.admin.core.scheduler.MisfireStrategyEnum;
import com.xxl.job.admin.core.scheduler.ScheduleTypeEnum;
import com.xxl.job.admin.core.thread.JobScheduleHelper;
import com.xxl.job.admin.core.thread.JobTriggerPoolHelper;
import com.xxl.job.admin.core.trigger.TriggerTypeEnum;
import com.xxl.job.admin.core.util.I18nUtil;
import com.xxl.job.admin.dao.XxlJobGroupDao;
import com.xxl.job.admin.dao.XxlJobInfoDao;
import com.xxl.job.admin.service.LoginService;
import com.xxl.job.admin.service.XxlJobService;
import com.xxl.job.core.biz.model.ReturnT;
import com.xxl.job.core.enums.ExecutorBlockStrategyEnum;
import com.xxl.job.core.glue.GlueTypeEnum;
import com.xxl.job.core.util.DateUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.sxp.common.utils.PageInfo;
import org.sxp.common.utils.PageResult;
import org.sxp.common.utils.Query;
import org.sxp.common.web.WrapperResponse;
import org.sxp.modules.dataworks.utils.XxlJobLogCaptureUtil;
import org.sxp.modules.sys.controller.AbstractController;
import org.sxp.modules.sys.shiro.ShiroUtils;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.*;

/**
 * index controller
 * @author xuxueli 2015-12-19 16:13:16
 */
@Controller
@RequestMapping("/jobinfo")
public class JobInfoController extends AbstractController {
	private static Logger logger = LoggerFactory.getLogger(JobInfoController.class);

	@Resource
	private XxlJobGroupDao xxlJobGroupDao;
	@Resource
	private XxlJobService xxlJobService;
	@Resource
	private XxlJobInfoDao xxlJobInfoDao;
	
	@RequestMapping
	public String index(HttpServletRequest request, Model model, @RequestParam(required = false, defaultValue = "-1") int jobGroup) {

		// 枚举-字典
		model.addAttribute("ExecutorRouteStrategyEnum", ExecutorRouteStrategyEnum.values());	    // 路由策略-列表
		model.addAttribute("GlueTypeEnum", GlueTypeEnum.values());								// Glue类型-字典
		model.addAttribute("ExecutorBlockStrategyEnum", ExecutorBlockStrategyEnum.values());	    // 阻塞处理策略-字典
		model.addAttribute("ScheduleTypeEnum", ScheduleTypeEnum.values());	    				// 调度类型
		model.addAttribute("MisfireStrategyEnum", MisfireStrategyEnum.values());	    			// 调度过期策略

		// 执行器列表
		List<XxlJobGroup> jobGroupList_all =  xxlJobGroupDao.findAll();

		// filter group
		List<XxlJobGroup> jobGroupList = filterJobGroupByRole(request, jobGroupList_all);
		if (jobGroupList==null || jobGroupList.size()==0) {
			throw new XxlJobException(I18nUtil.getString("jobgroup_empty"));
		}

		model.addAttribute("JobGroupList", jobGroupList);
		model.addAttribute("jobGroup", jobGroup);

		return "modules/xxljob/jobinfo";
	}

	public static List<XxlJobGroup> filterJobGroupByRole(HttpServletRequest request, List<XxlJobGroup> jobGroupList_all){
		List<XxlJobGroup> jobGroupList = new ArrayList<>();
		if (jobGroupList_all!=null && jobGroupList_all.size()>0) {
//			if (ShiroUtils.isAdmin()) {
				jobGroupList = jobGroupList_all;
//			} else {
//				List<String> groupIdStrs = new ArrayList<>();
//				if (loginUser.getPermission()!=null && loginUser.getPermission().trim().length()>0) {
//					groupIdStrs = Arrays.asList(loginUser.getPermission().trim().split(","));
//				}
//				for (XxlJobGroup groupItem:jobGroupList_all) {
//					if (groupIdStrs.contains(String.valueOf(groupItem.getId()))) {
//						jobGroupList.add(groupItem);
//					}
//				}
//			}
		}
		return jobGroupList;
	}

	public ReturnT updateJobParam(Integer jobinfoId, String param){
		ReturnT r = null;
		ReturnT<XxlJobInfo> returnT = loadById(jobinfoId);
		if(returnT.getCode() == ReturnT.SUCCESS_CODE) {
			XxlJobInfo jobInfo = returnT.getContent();
			if(jobInfo != null){
				jobInfo.setExecutorParam(param);
				r = update(jobInfo);
			}else{
				r = new ReturnT<String>("任务不存在，请重新配置!!!");
			}
		}else{
			r = returnT;
		}
		return r;
	}

	public static void validPermission(HttpServletRequest request, int jobGroup) {
		XxlJobUser loginUser = (XxlJobUser) request.getAttribute(LoginService.LOGIN_IDENTITY_KEY);
		if (!loginUser.validPermission(jobGroup)) {
			throw new RuntimeException(I18nUtil.getString("system_permission_limit") + "[username="+ loginUser.getUsername() +"]");
		}
	}
	
	@RequestMapping("/pageList")
	@ResponseBody
	public WrapperResponse pageList(@RequestParam(required = false, defaultValue = "-1") int jobGroup,
									@RequestParam(required = false, defaultValue = "-1") int triggerStatus,
									String jobDesc, String executorHandler, String author) {
		PageInfo pageInfo = buildPageInfo();
		// xxljob默认从0开始与mybatisplus不同处理一下
		pageInfo.setPageNum(pageInfo.getPageNum()-1);
		Map<String, Object> resultMap = xxlJobService.pageList(pageInfo.getPageNum()*pageInfo.getPageSize(), pageInfo.getPageSize(), jobGroup, triggerStatus, jobDesc, executorHandler, author);
		// xxljob默认从0开始，上面减1这里加回去
		pageInfo.setPageNum(pageInfo.getPageNum()+1);

		PageResult<XxlJobInfo> pageResult = new PageResult();
		pageResult.setRecordCounts(Integer.parseInt(resultMap.get("recordsTotal").toString()));
		pageResult.setPages(Query.countPageNum(pageResult.getRecordCounts(), pageInfo.getPageSize()));
		pageResult.setData((List<XxlJobInfo>) resultMap.get("data"));
		pageResult.setPageNum(pageInfo.getPageNum());
		pageResult.setPageSize(pageInfo.getPageSize());

		return WrapperResponse.success(pageResult);
	}
	
	@RequestMapping("/add")
	@ResponseBody
	public ReturnT<String> add(@RequestBody XxlJobInfo jobInfo) {
		if(StrUtil.isNotBlank(jobInfo.getAlarmEmail())){
			Set<String> emailSet = new HashSet<String>(Arrays.asList(jobInfo.getAlarmEmail().split(",")));
			for (String email : emailSet) {
				boolean isEmail = Validator.isEmail(email);
				if(!isEmail){
					return new ReturnT<>(ReturnT.FAIL_CODE, "报警邮件格式不正确!");
				}
			}
		}

		return xxlJobService.add(jobInfo);
	}

	@RequestMapping("/loadById")
	@ResponseBody
	public ReturnT<XxlJobInfo> loadById(int id){
		XxlJobInfo jobInfo = xxlJobInfoDao.loadById(id);
		return new ReturnT<XxlJobInfo>(jobInfo);
	}
	
	@RequestMapping("/update")
	@ResponseBody
	public ReturnT<String> update(@RequestBody XxlJobInfo jobInfo) {
		if(StrUtil.isNotBlank(jobInfo.getAlarmEmail())){
			Set<String> emailSet = new HashSet<String>(Arrays.asList(jobInfo.getAlarmEmail().split(",")));
			for (String email : emailSet) {
				boolean isEmail = Validator.isEmail(email);
				if(!isEmail){
					return new ReturnT<>(ReturnT.FAIL_CODE, "报警邮件格式不正确!");
				}
			}
		}
		return xxlJobService.update(jobInfo);
	}
	
	@RequestMapping("/remove")
	@ResponseBody
	public ReturnT<String> remove(int id) {
		return xxlJobService.remove(id);
	}
	
	@RequestMapping("/stop")
	@ResponseBody
	public ReturnT<String> pause(int id) {
		return xxlJobService.stop(id);
	}
	
	@RequestMapping("/start")
	@ResponseBody
	public ReturnT<String> start(int id) {
		return xxlJobService.start(id);
	}
	
	@RequestMapping("/trigger")
	@ResponseBody
	//@PermissionLimit(limit = false)
	public ReturnT<String> triggerJob(int id, String executorParam, String addressList) {
		// force cover job param
		if (executorParam == null) {
			executorParam = "";
		}
		JobTriggerPoolHelper.trigger(id, TriggerTypeEnum.MANUAL, -1, null, executorParam, addressList);
		return ReturnT.SUCCESS;
	}

	@RequestMapping("/nextTriggerTime")
	@ResponseBody
	public ReturnT<List<String>> nextTriggerTime(String scheduleType, String scheduleConf) {

		XxlJobInfo paramXxlJobInfo = new XxlJobInfo();
		paramXxlJobInfo.setScheduleType(scheduleType);
		paramXxlJobInfo.setScheduleConf(scheduleConf);

		List<String> result = new ArrayList<>();
		try {
			Date lastTime = new Date();
			for (int i = 0; i < 5; i++) {
				lastTime = JobScheduleHelper.generateNextValidTime(paramXxlJobInfo, lastTime);
				if (lastTime != null) {
					result.add(DateUtil.formatDateTime(lastTime));
				} else {
					break;
				}
			}
		} catch (Exception e) {
			logger.error(e.getMessage(), e);
			return new ReturnT<List<String>>(ReturnT.FAIL_CODE, (I18nUtil.getString("schedule_type")+I18nUtil.getString("system_unvalid")) + e.getMessage());
		}
		return new ReturnT<List<String>>(result);

	}
	
}
